# ----- model general -----#
name = "model_2"
mode = "debug"
check_stats = True
region = "henan"
feature = "optical_vi"
save_folder = f"/your_path_to_save_checkpoints/{region}/{name}"
data_root = f"/your_path_to_load_pickles/{region}"
description = f"the no.{name.split('_')[-1]} model for {region} that used {feature}"
track_progress = False
# ----- model general -----#

# ----- model hyperparameter -----#
batch_size = 1000
max_lr = 3e-4
div_factor = 10
pct_start = 0.2
delta_date = 30
num_epoch = 500
num_layer = 7
customized_weight = False
weight = [1, 2, 3, 4]
# ----- model hyperparameter -----#

# ----- remote sensing trick -----#
metrics = ["evi"]
COI_DICT = {
    "heilongjiang": ["negative", "corn", "soybeans", "rice"],
    "liaoning": ["negative", "corn", "soybeans", "rice"],
    "hubei": ["negative", "rice"],
    "innermongolia": ["negative", "corn", "soybeans"],
    "xinjiang": ["negative", "cotton"],
    "henan": ["negative", "corn"],
    "shandong": ["negative", "corn"],
}
class_of_interests = COI_DICT[region]
interpolate = True
if feature == "optical":
    num_channel = 10
elif feature == "vi":
    num_channel = len(metrics)
else:
    num_channel = 10 + len(metrics)
# ----- remote sensing trick -----#

xinjiang = {
    "天山区": 3,
    "水磨沟区": 3,
    "玛纳斯县": 3,
    "沙湾市": 3,
    "石河子市": 3,
    "察布查尔锡伯自治县": 3,
    "奎屯市": 3,
    "乌苏市": 3,
    "精河县": 3,
    "胡杨河市": 3,
}

hubei = {"湖北省": 1}

henan = {
    "河南省": 1,
}
shandong = {
    "山东省": 1,
}
innerMongolia = {
    "科尔沁左翼后旗": 3,
    "科尔沁区": 3,
    "开鲁县": 3,
    "奈曼旗": 3,
    "敖汉旗": 3,
    "松山区": 3,
    "元宝山区": 3,
    "红山区": 3,
    "达拉特旗": 3,
    "杭锦旗": 3,
    "鄂托克旗": 3,
    "乌审旗": 3,
    "扎兰屯市": 3,
    "阿荣旗": 3,
    "扎赉特旗": 3,
    "科尔沁右翼前旗": 3,
    "正蓝旗": 3,
    "正镶白旗": 3,
    "镶黄旗": 3,
    "科尔沁左翼中旗": 3,
    "甘南县": 3,
    "莫力达瓦达斡尔族自治旗": 3,
    "鄂伦春自治旗": 3,
    "乌兰浩特市": 3,
    "科尔沁右翼中旗": 3,
    "牙克石市": 3,
    "突泉县": 3,
    "太仆寺旗": 3,
}
heilongjiang = {"黑龙江省": 1}

liaoning = {"辽宁省": 1}
